package com.other;

import java.util.Set;
import java.util.TreeSet;

/**
 * @author yyh
 * @date 2024年11月11日 下午4:31
 */
public class HappyNumber {
    /**
     * 编写一个算法来判断一个数 n 是不是快乐数。
     *
     * 「快乐数」 定义为：
     *
     * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
     * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
     * 如果这个过程 结果为 1，那么这个数就是快乐数。
     * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
     *
     * 示例 1：
     *
     * 输入：n = 19
     * 输出：true
     * 解释：
     * 12 + 92 = 82
     * 82 + 22 = 68
     * 62 + 82 = 100
     * 12 + 02 + 02 = 1
     * 示例 2：
     *
     * 输入：n = 2
     * 输出：false
     */

    public boolean isHappy(int n) {
        Set<Integer> set = new TreeSet<>();
        int x = n;
        set.add(x);
        while((x = calculate(x)) != 1){
            if(set.contains(x)){
                return false;
            }
            set.add(x);
        }
        return true;
    }

    private int calculate(int n){
        int r = 0;
        while (n >=10){
            int d = n%10;
            n = n/10;
            r += d*d;
        }
        r += n*n;
        return r;
    }

    public static void main(String[] args) {
        HappyNumber hp = new HappyNumber();
        System.out.println(hp.isHappy(1));
        System.out.println(hp.isHappy(2));
        System.out.println(hp.isHappy(44));
        System.out.println(hp.isHappy(Integer.MAX_VALUE));
    }
}
